from common.base_page import BasePage
from po.product_add_page import ProductAddPage
from po.product_list_page import ProductListPage


class MainPage(BasePage):
    def goto_product_manage(self):  # 进入商品管理菜单
        self.click_element(locator=self['product_manage'])
        return ProductAddPage()

    def goto_product_list(self):  # 进入商品列表页
        self.click_element(locator=self['product_list'])
        return ProductListPage()
